//
//  UIButton+CountDown.m
//  HSCategoryKit
//
//  Created by huang on 2018/8/1.
//

#import "UIButton+CountDown.h"

@implementation UIButton (CountDown)

- (void)startTime:(NSInteger )timeout title:(NSString *)title waitTittle:(NSString *)waitTittle
{
    __block NSInteger timeOut = timeout; //倒计时时间
    dispatch_queue_t queue = dispatch_get_global_queue(DISPATCH_QUEUE_PRIORITY_DEFAULT, 0);
    dispatch_source_t _timer = dispatch_source_create(DISPATCH_SOURCE_TYPE_TIMER, 0, 0,queue);
    dispatch_source_set_timer(_timer,dispatch_walltime(NULL, 0),1.0*NSEC_PER_SEC, 0); //每秒执行
    dispatch_source_set_event_handler(_timer, ^{
        if(timeOut <= 0){ //倒计时结束，关闭
            dispatch_source_cancel(_timer);
            dispatch_async(dispatch_get_main_queue(), ^{
                //设置界面的按钮显示 根据自己需求设置
                [self setTitle:title forState:UIControlStateNormal];
                self.enabled = YES;
            });
        }else{
            NSString *strTime = [NSString stringWithFormat:@"%ld", (long)timeOut];
            dispatch_async(dispatch_get_main_queue(), ^{
                //设置界面的按钮显示 根据自己需求设置
                [self setTitle:[NSString stringWithFormat:@"%@%@",strTime,waitTittle] forState:UIControlStateDisabled];
                self.enabled = NO;
                
            });
            timeOut--;
            
        }
    });
    dispatch_resume(_timer);
}

@end
